package com.spzx.common.rabbit.service;

import com.alibaba.fastjson2.JSON;
import com.spzx.common.core.utils.uuid.UUID;
import com.spzx.common.rabbit.entity.GuiguCorrelationData;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.concurrent.TimeUnit;

@Service
public class RabbitService {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 发送消息
     *
     * @param exchange   交换机
     * @param routingKey 路由键
     * @param message    消息
     */
    public boolean sendMessage(String exchange, String routingKey, Object message) {
        String uuid = UUID.randomUUID().toString().replaceAll("-","");

        //解决消息重发问题，那么发消息时需要携带关联数据，以后根据关联数据进行消息重发
        GuiguCorrelationData guiguCorrelationData = new GuiguCorrelationData();
        guiguCorrelationData.setId(uuid);
        guiguCorrelationData.setExchange(exchange);
        guiguCorrelationData.setRoutingKey(routingKey);
        guiguCorrelationData.setMessage(message);

        //将关联数据存储在redis中。重发消息时，可以从redis中获取关联数据进行消息重发。
        //设置合理失效时间：这个时间需要大于重试消息。
        redisTemplate.opsForValue().set(uuid, JSON.toJSONString(guiguCorrelationData),10, TimeUnit.MINUTES);

        //必须要先保存缓存数据再发消息。
        rabbitTemplate.convertAndSend(exchange, routingKey, message,guiguCorrelationData);

        return true;
    }

    /**
     * 封装发送延迟消息的方法
     * @param exchangeDelay 交换机名称
     * @param routingDelay 路由key名称
     * @param msg  消息
     * @param delayTime 延迟时间
     */
    public void sendDealyMessage(String exchangeDelay, String routingDelay, Object msg, int delayTime) {

        String uuid = UUID.randomUUID().toString().replaceAll("-","");

        //解决消息重发问题，那么发消息时需要携带关联数据，以后根据关联数据进行消息重发
        GuiguCorrelationData guiguCorrelationData = new GuiguCorrelationData();
        guiguCorrelationData.setId(uuid);
        guiguCorrelationData.setExchange(exchangeDelay);
        guiguCorrelationData.setRoutingKey(routingDelay);
        guiguCorrelationData.setMessage(msg);
        guiguCorrelationData.setDelay(true); //延迟消息设置为true
        guiguCorrelationData.setDelayTime(delayTime); //单位秒

        //将关联数据存储在redis中。重发消息时，可以从redis中获取关联数据进行消息重发。
        //设置合理失效时间：这个时间需要大于重试消息。
        redisTemplate.opsForValue().set(uuid, JSON.toJSONString(guiguCorrelationData),10, TimeUnit.MINUTES);

        MessagePostProcessor messagePostProcessor = (message)->{
            message.getMessageProperties().setDelay(delayTime*1000); //延迟时间10秒。单位毫秒数
            return message;
        };
        rabbitTemplate.convertAndSend(exchangeDelay, routingDelay, msg,messagePostProcessor,guiguCorrelationData);
    }
}